package com.rabbitmq.config;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.nio.charset.StandardCharsets;

/**
 * @Author： yu.Tan
 * @Date: 2020/8/20 14:26
 * @Description: 自定义消息发送确认的回调
 *  实现接口：implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback
 *  ConfirmCallback：只确认消息是否正确到达交换机中,不管是否到达交换机,该回调都会执行
 *  ReturnCallback：如果消息从交换机未正确到达队列中将会执行，正确到达则不执行;
 *
 */
@Component
public class CustomConfirmAndReturnCallback implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnCallback {


    private static final Logger logger = LoggerFactory.getLogger(CustomConfirmAndReturnCallback.class);

    @Autowired
    private RabbitTemplate rabbitTemplate;


    /**
     * PostConstruct: 用于在依赖关系注入完成之后需要执行的方法上，以执行任何初始化.
     */
    @PostConstruct
    public void init(){

        //指定 ConfirmCallback
        rabbitTemplate.setConfirmCallback(this);

        //指定 ReturnCallback
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     *
     * @param correlationData 可以封装业务ID信息，需要在发送消息时传入此参数，这里才能接收到，否则是null
     * @param b 消息发送的结果状态，发送成功是true，失败是false
     * @param s 发送失败的描述信息，如果发送成功是null。
     *  这个接口的作用是生产者把消息发送到交换机的结果回调。(Producer ----->  Exchange)
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean b, String s) {
        logger.info("confirm回调方法>>>回调消息ID为: " + correlationData.getId());
        if (b) {
            logger.info("confirm回调方法>>>消息发送到交换机成功！");
        } else {
            logger.info("confirm回调方法>>>消息发送到交换机失败！，原因 : [{}]", s);
        }
    }

    /**
     * 如果消息没有到达交换机,则该方法中isSendSuccess = false,error为错误信息;
     * 如果消息正确到达交换机,则该方法中isSendSuccess = true;
     * @param message 发送的消息内容 + 发送消息的配置信息
     * @param i 状态码，发送成功是200
     * @param s 发送消息失败的描述信息
     * @param s1 消息使用的交换机
     * @param s2 消息使用的路由键
     * 这个接口的作用是消息从交换机到队列的结果回调。(Exchange  -----> Queue )。
     * 消息从交换到队列失败，失败原因可能是路由键不存在，通道未绑定等等，一般都跟配置有关系
     */
    @Override
    public void returnedMessage(Message message, int i, String s, String s1, String s2) {
        logger.info("returnedMessage回调方法>>>" + new String(message.getBody(), StandardCharsets.UTF_8) + ",replyCode:" + i
                + ",replyText:" +  s + ",exchange:" + s1 + ",routingKey:" + s2);
    }
}
